package com.itheima.manage.interceptor;

import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.SecureUtil;
import cn.hutool.json.JSONUtil;
import com.itheima.manage.config.MyCache;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Map;

/**
 * @author yaohu
 * @date 2021/12/8
 * @Version 1.0
 * @describe
 */
@Component
public class CacheInterceptor implements HandlerInterceptor {

    @Autowired
    private RedisTemplate redisTemplate;

    /**
     * 请求到达目标资源之前 可以增强request
     *
     * @param request
     * @param response
     * @param handler
     * @return
     * @throws Exception
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

        // 如果请求中调用的不是controller中的方法 直接放行
        if (!(handler instanceof HandlerMethod)) {
            return true;
        }

        // 如果不是GET请求直接放行
        if (!((HandlerMethod) handler).hasMethodAnnotation(GetMapping.class)) {
            return true;
        }

        // 如果没有MyCache注解直接放行
        if (!((HandlerMethod) handler).hasMethodAnnotation(MyCache.class)) {
            return true;
        }

        // 从redis中获取缓存数据
        String cacheData = (String) redisTemplate.opsForValue().get(getCacheKey(request));
        if (StrUtil.isBlank(cacheData)) {
            // 请求还没有缓存 直接放行
            return true;
        }
        System.out.println("数据从缓存中取走了...");

        response.setContentType("application/json;charset=utf-8");
        response.getWriter().println(cacheData);
        return false;
    }

    public static String getCacheKey(HttpServletRequest request) {
        // 获取请求地址
        String requestURI = request.getRequestURI();
        // 获取请求参数
        Map<String, String[]> parameterMap = request.getParameterMap();
        // 获取请求Token
        String token = request.getHeader("Authorization");
        // 返回一个唯一的redis的key
        return SecureUtil.md5(requestURI + "_" + JSONUtil.toJsonStr(parameterMap) + "_" + token);
    }
}
